%run train-cGAN # %run unit_test
self.device is cuda:0 64 Epoch: 0, Itration: 0 loss_G_GAN = 5.2222394943237305, loss_G_L1 = 0.42152929306030273 * 5.0 loss_D = 0.6994704008102417
Epoch: 76, Itration: 1000 loss_G_GAN = 1.7559757232666016, loss_G_L1 = 0.05914055183529854 * 5.0 loss_D = 0.32040828466415405
Epoch: 153, Itration: 2000 loss_G_GAN = 2.0574169158935547, loss_G_L1 = 0.05481714755296707 * 5.0 loss_D = 0.2651404142379761
save model to ./checkpoints/pix2food.pkl at iteration = 3000 Epoch: 230, Itration: 3000 loss_G_GAN = 1.2604695558547974, loss_G_L1 = 0.04347262904047966 * 5.0 loss_D = 0.3395516872406006
Finally save model to ./checkpoints/pix2food.pkl # --- Validation Set --- #
# %run train-cGAN
# %run train